exp_id="mynet_plus"

backbone='resnet50'
dataset='vcoco'
dataset_path='data/v-coco'
pretrained='logs/vcoco/mynet_plus/stage1/checkpoint_best.pth'
source_dir=$dataset_path/images/val2014

# python plot_ours_feature.py \
# --pretrained $pretrained \
# --dataset_file $dataset \
# --hoi_path$dataset_path \
# --source_dir $dataset_path/images/test2015 \
# --num_obj_classes 80 \
# --num_verb_classes 117 \
# --backbone $backbone \
# --num_queries 64 \
# --dec_layers_hopd 6 \
# --dec_layers_interaction 6 \
# --top_k 5


python plot_ours_feature.py \
--pretrained $pretrained \
--dataset_file $dataset \
--hoi_path $dataset_path \
--output_dir picture/$dataset/$exp_id/ \
--source_dir $source_dir \
--num_obj_classes 81 \
--num_verb_classes 29 \
--backbone $backbone \
--num_queries 100 \
--dec_layers_hopd 3 \
--dec_layers_interaction 3 \
--top_k 5